package com.example.chamberlainserver.mapper;

import com.example.chamberlainserver.Entry.ScheduleAssignment;
import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Param;

import java.time.LocalDate;
import java.util.List;
import java.util.Map;

/**
 * 排班分配数据访问接口
 */
@Mapper
public interface ScheduleAssignmentMapper {
    /**
     * 根据ID查询排班分配
     */
    ScheduleAssignment getScheduleAssignmentById(@Param("id") String id);
    
    /**
     * 根据用户ID查询排班分配
     */
    List<ScheduleAssignment> getScheduleAssignmentsByUserId(@Param("userId") Integer userId);
    
    /**
     * 根据日期范围查询排班分配
     */
    List<ScheduleAssignment> getScheduleAssignmentsByDateRange(
        @Param("startDate") LocalDate startDate, 
        @Param("endDate") LocalDate endDate
    );
    
    /**
     * 根据班次ID查询排班分配
     */
    List<ScheduleAssignment> getScheduleAssignmentsByShiftId(@Param("shiftId") String shiftId);
    
    /**
     * 根据用户ID和日期查询排班分配
     */
    List<ScheduleAssignment> getScheduleAssignmentsByUserAndDate(
        @Param("userId") Integer userId, 
        @Param("scheduleDate") LocalDate scheduleDate
    );
    
    /**
     * 新增排班分配
     */
    int insertScheduleAssignment(ScheduleAssignment scheduleAssignment);
    
    /**
     * 批量新增排班分配
     */
    int batchInsertScheduleAssignment(@Param("list") List<ScheduleAssignment> list);
    
    /**
     * 更新排班分配
     */
    int updateScheduleAssignment(ScheduleAssignment scheduleAssignment);
    
    /**
     * 删除排班分配
     */
    int deleteScheduleAssignmentById(@Param("id") String id);
    
    /**
     * 更新排班分配状态
     */
    int updateScheduleAssignmentStatus(@Param("id") String id, @Param("status") String status);
    
    /**
     * 分页查询排班分配列表
     */
    List<ScheduleAssignment> getPagedScheduleAssignmentList(Map<String, Object> params);
    
    /**
     * 获取分页查询的总记录数
     */
    Long getPagedScheduleAssignmentCount(Map<String, Object> params);
    
    /**
     * 统计用户在指定时间范围内的排班次数
     */
    Integer countUserAssignmentsInDateRange(
        @Param("userId") Integer userId,
        @Param("startDate") LocalDate startDate,
        @Param("endDate") LocalDate endDate
    );
}